Latent On-Policy Self-Distillation
潜在在线策略自蒸馏
🧭 LLM Agent 在线策略自蒸馏训练 ★ 深度解读
National University of Singapore · Beijing University of Posts and Telecommunications · Shanghai Jiao Tong University
让 OPSD 的特权上下文从手工设计变为可学习的潜在表示,联合蒸馏提升 agent 与代码策略,样本效率显著优于 GRPO/Skill-SD
📌 工作参考价值(4 点)
- 在电商导购/客服 Agent 训练中,可将历史成功会话(用户目标+操作序列+结果)检索并压缩成 latent tokens 条件化 teacher,让学生从自己的 rollout 中稠密学习,避免人工写死“成功策略/提示模板”;线上推理只保留学生,不增加延迟。
- 使用 on-policy self-distillation 时必须加 privileged-margin 约束,否则 teacher 会退化到 student,联合训练效果反而比冻结 composer 更差;可用结果奖励(如转化/点击)加权 per-token advantage,margin 建议设在 0.05 左右。
- latent context 注入生产推理引擎不必改底层 embedding,用 placeholder sentinel + dummy token ids 即可兼容 vLLM/SGLang;训练后 composer 和检索索引可以完全离线。
- 在 RLVR 采样预算有限(如每天 1-2k 条 rollout)且需要快速策略迭代的推荐/广告 Agent 场景,latent-teacher 稠密监督比纯 GRPO 样本效率高很多,可作为内部训练组件复用。